kNN Algorithm using Python